dataset/training/main_generator.py